import unittest
import random


def my_sum(a, b):
    return a + b


def my_random():  # 返回从1到5之间的随机数
    return random.randint(1, 5)


class my_test(unittest.TestCase):

    def test_001(self):
        num1 = my_sum(3, 4)  # 通过定义变量num1 得到my_sum的返回值
        self.assertEqual(num1, 7)  # num1里存放实际结果,7是预期结果
        # 实际结果与预期结果相符,表示测试用例通过
        # 实际结果与预期结果相符,表示测试用例失败

    def test_002(self):
        num2 = my_sum(5, 6)
        self.assertEqual(num2, 11)

    def test_003(self):
        num3 = my_random()
        self.assertIn(num3, [1, 2, 3, 4, 5])
